<?php

/**
 * @file
 * Views handler: "Add to cart" form as a field.
 */

/**
 * Displays the "Add to cart" form like the product page.
 */
class uc_product_handler_field_addtocart extends views_handler_field {

    /**
     * Overrides views_handler_field::query().
     */
    function query() {
        $this->ensure_my_table();
        $this->add_additional_fields();
    }

    /**
     * Overrides views_handler_field::element_type().
     */
    function element_type() {
        if (isset($this->definition['element type'])) {
            return $this->definition['element type'];
        }
        return 'div';
    }

    /**
     * Overrides views_handler_field::render().
     */
    function render($values) {
        if (uc_product_is_product($values->{$this->aliases['type']})) {
            $type = node_get_types('type', $values->{$this->aliases['type']});
            $module = $type->module;
            $product = node_load($values->{$this->aliases['nid']});
            if (function_exists('theme_' . $module . '_add_to_cart')) {
                return theme($module . '_add_to_cart', $product);
            } elseif (function_exists('theme_uc_product_add_to_cart')) {
                return theme('uc_product_add_to_cart', $product);
            }
        }
    }

}
